import axios, { AxiosRequestConfig, Canceler } from "axios";
import qs from "qs";

// 申明 Map, 存请求标识和取消函数
let pendingMap = new Map<string, Canceler>();

export const getPendingUrl = (config: AxiosRequestConfig) => {
  return [
    config.method,
    config.url,
    qs.stringify(config.data),
    qs.stringify(config.params),
  ].join("&");
};

/**
 * @description 取消axios请求的类.
 * 本文件用作取消请求, 在某个请求还未完成而跳转到新的页面时, 需要把前面的请求给取消掉, 以释放资源.
 */
export class AxiosCanceler {
  /**
   * @description 添加请求到Map中
   * @param config
   */
  addPending(config: AxiosRequestConfig) {
    // 取消请求, 且如果Map中已有请求内容, 移除.
    this.removePending(config);
    const url = getPendingUrl(config);
    config.cancelToken =
      config.cancelToken ||
      new axios.CancelToken((cancel: Canceler) => {
        if (!pendingMap.has(url)) {
          // Map中存入新的请求内容.
          pendingMap.set(url, cancel);
        }
      });
  }

  /**
   * @description 取消请求, 并从Map中移除.
   * @param config
   */
  removePending(config: AxiosRequestConfig) {
    const url = getPendingUrl(config);

    if (pendingMap.has(url)) {
      const cancel = pendingMap.get(url);
      // 取消请求, 改写法保证当且仅当前面为true时才执行后面的代码.
      cancel && cancel(url);
      pendingMap.delete(url);
    }
  }

  /**
   * @description 取消所有的请求, 并从Map中移除.
   */
  removeAllPending() {
    // 取消所有请求.
    pendingMap.forEach((cancel: Canceler) => {
      cancel && cancel();
    });
    // 从Map中移除.
    pendingMap.clear();
  }

  /**
   * @description 重置Map.
   */
  reset(): void {
    pendingMap = new Map<string, Canceler>();
  }
}
